export dataset_name=data/proteingym_v2/DATASET

CUDA_VISIBLE_DEVICES=2 python src/Prime/predict.py \
    --dataset_input $dataset_name \
    --model_location data/esm_meer \
    --mutation_col mutant